package com.gitee.mp.core.mapper;

import cn.hutool.core.collection.CollUtil;
import com.gitee.mp.domain.condition.ConditionBuilder;

import java.util.Collections;
import java.util.List;

/**
 * @author hongda.li
 */
public interface MilvusMapper<Data> {
    /**
     * 新增数据
     *
     * @param data 待新增数据
     * @return 新增数量
     */
    default long insert(Data data) {
        return this.batchInsert(Collections.singletonList(data));
    }

    /**
     * 根据主键更新数据
     * 若数据不存在则插入
     *
     * @param data 待更新数据
     * @return 更新数量
     */
    default long upsert(Data data) {
        return this.batchUpsert(Collections.singletonList(data));
    }

    /**
     * 删除数据
     * 根据主键删除
     *
     * @param data 待删除数据
     * @return 删除数量
     */
    default long delete(Data data) {
        return this.batchDelete(Collections.singletonList(data));
    }

    /**
     * 批量新增数据
     *
     * @param dataList 待新增数据集合
     * @return 新增数量
     */
    long batchInsert(List<Data> dataList);

    /**
     * 根据主键批量更新数据
     * 若数据不存在则插入
     *
     * @param dataList 待更新数据集合
     * @return 更新数量
     */
    long batchUpsert(List<Data> dataList);

    /**
     * 批量删除数据
     * 根据主键删除
     *
     * @param dataList 待删除数据集合
     * @return 删除数量
     */
    long batchDelete(List<Data> dataList);

    /**
     * 批量删除
     * 根据条件删除
     *
     * @param builder 条件构建器
     * @return 删除数量
     */
    long batchDelete(ConditionBuilder<Data> builder);

    /**
     * 按照条件查询第一条数据
     * 支持标量检索与向量检索
     *
     * @param builder 条件构建器
     * @return 检索结果
     */
    default Data selectOne(ConditionBuilder<Data> builder) {
        return CollUtil.getFirst(this.select(builder));
    }

    /**
     * 按照条件查询数据
     * 支持分页查询
     * 支持标量检索与向量检索
     *
     * @param builder 条件构建器
     * @return 检索结果
     */
    List<Data> select(ConditionBuilder<Data> builder);
}
